Skip to content

Handle CUDA NMS fallback in YOLO11 plugin#4

Open
mathieu-thomas wants to merge 1 commit intomainfrom
codex/explain-how-plugins-work-0fgome
Open

Handle CUDA NMS fallback in YOLO11 plugin#4
mathieu-thomas wants to merge 1 commit intomainfrom
codex/explain-how-plugins-work-0fgome

Conversation

@mathieu-thomas
Copy link
Owner

Summary

  • detect when torchvision lacks CUDA NMS support and hide GPU device options accordingly
  • add runtime fallback to CPU inference when CUDA NMS errors occur for image, video, or webcam pipelines
  • update plugin UI defaults and version metadata to reflect safer device handling

Testing

  • python -m compileall plugins/wan2gp-yolo11

Codex Task

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant